from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework.permissions import IsAdminUser
from datetime import date, timedelta
from users.models import User
from goods.models import GoodsVisitCount


# 用户总量统计
class UserTotalCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取注册用户总数（普通用户）
        count = User.objects.filter(is_staff=False).count()
        # 返回数量
        return Response({'count': count})


# 日增用户统计
class UserDayCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取当日注册用户数量 date_joined 记录创建账户时间
        count = User.objects.filter(is_staff=False, date_joined__gte=now_date).count()
        return Response({'count': count})


# 日活跃用户统计
class UserActiveCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取当日登录用户数量  last_login记录最后登录时间
        count = User.objects.filter(is_staff=False, last_login__gte=now_date).count()
        return Response({'count': count})


# 日下单用户量统计
class UserOrderCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取当日下单用户数量  orders__create_time 订单创建时间
        user = User.objects.filter(is_staff=False, orders__create_time__gte=now_date)
        # 去重
        users = set(user)
        # 判定长度
        count = len(users)
        return Response({'count': count})


# 日分类商品访问量月增用户统计
class UserMonthCountView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取一个月前日期
        old_date = now_date - timedelta(30)
        # 创建空列表保存每天的用户量
        date_list = []
        # 循环遍历获取当天日期
        for i in range(31):
            # 起始日期
            index_date = old_date + timedelta(i)
            # 起始日期的下一天
            next_date = old_date + timedelta(i + 1)
            # 查询条件是大于当前日期index_date，小于明天日期的用户cur_date，得到当天用户量
            count = User.objects.filter(is_staff=False, date_joined__gte=index_date, date_joined__lt=next_date).count()
            date_list.append({'count': count, 'date': index_date})
        # 返回数量
        return Response(date_list)


# 日分类商品访问量
class GoodsDayView(APIView):
    # 指定管理员权限
    permission_classes = [IsAdminUser]

    def get(self, request):
        # 获取当前日期
        now_date = date.today()
        # 获取当天访问的商品分类数量信息
        goods = GoodsVisitCount.objects.filter(date__gte=now_date)
        date_list = []
        for good in goods:
            # 获取单个商品分类数量信息
            count = good.count
            # 获取关联对象的名字
            category = good.category.name
            date_list.append({'count': count, 'category': category})
        # 返回数量
        return Response(date_list)
